package com.atnjupt.servlet;
/**
 * @author Wizarder
 * @create 2021-06-03 12:36
 */


import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;

public class DownLoad extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //1 获取下载的文件名
        String downloadName = "11.jpg";
        //2 读取下载的文件的内容(通过ServletContext对象可以读取)
        ServletContext servletContext = getServletContext();
        String mimeType = servletContext.getMimeType("/file/" + downloadName);
        System.out.println("下载文件的类型：" + mimeType);
        // 4、在回传前，通过响应头告诉客户端返回的数据类型
        response.setContentType(mimeType);
        //5、还要告诉客户端收到的数据是用于下载使用（还是使用响应头）
//            response.setHeader("Content-Disposition","attachment;filename=" + downloadName);//设置响应头可以下载
        //如果要想更改下载文件为中文，需要进行如下的操作
        //判断浏览器类型，因为不同的浏览器类型中文字符编码不一样 火狐是base6,ie和谷歌是urlcode
        if(request.getHeader("User-Agent").contains("Firefox")){
        // 如果是火狐浏览器使用Base64编码
            response.setHeader("Content-Disposition","attachment;filename==?UTF-8?B?" + new BASE64Encoder().encode("美女.jpg".getBytes("UTF-8"))+"?=");
        }else{
            //如果是谷歌和ie
            response.setHeader("Content-Disposition","attachment;filename="+ URLEncoder.encode("美女.jpg","UTF-8"));
        }
        /**
         * /斜杠被服务器解析表示地址为http://ip:prot/工程名/  映射 到代码的Web目录
         */
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadName);
        // 获取响应的输出流
        ServletOutputStream outputStream = response.getOutputStream();
        //        3、把下载的文件内容回传给客户端
        // 读取输入流中全部的数据，复制给输出流，输出给客户端
        IOUtils.copy(resourceAsStream,outputStream);
    }


}
